import logging
from django.core.management import BaseCommand

from storage.rawdata import start_consume

_LOGGER = logging.getLogger(__name__)


class Command(BaseCommand):

    def add_arguments(self, parser):
        parser.add_argument('--tenant_name', dest='tenant_name', help='loki|witch|zs')

    def handle(self, **kwargs):
        tenant_name = kwargs['tenant_name']
        start_consume.start(tenant_name)
